# https://gitee.com/luo_zhi_cheng/TensorRT/tree/v10.9.0/tools/experimental/trt-engine-explorer/utils

# 报错Make sure to enable detailed ProfilingVerbosity when building the engine.，需要在构建engine时--profilingVerbosity=detailed
# https://github.com/NVIDIA/TensorRT/issues/2260

trt_exp_utils_dir=/opt/TensorRT/tools/experimental/trt-engine-explorer/utils
trt_exp_py_path=${trt_exp_utils_dir}/draw_engine.py

# graph_json_path=/sam_dev/sam_trt/temp/1/sam2.1_hiera_small_encoder_fp32.trt.graph.json
graph_json_path=/sam_dev/sam_trt/temp/1/sam2.1_hiera_small_encoder_fp16.trt.graph.json

python3 ${trt_exp_py_path} ${graph_json_path}

echo "在vscode中双击打开生成的svg文件就可以看模型的结构啦"
